fixing list object is not an iterator error